#创建一个浏览器对象
import datetime
import os
import sys
from selenium import webdriver
from selenium.webdriver.edge.service import Service

#创建类
class Driver:

    # 驱动路径
    path = "..\\msedgedriver.exe"
    driver = ""
    def __init__(self):
        options = webdriver.EdgeOptions()
        self.driver = webdriver.Edge(service=Service(self.path),options=options)

    def SaveScreenCut(self,name):
        """
        用于保存屏幕截图
        :return:
        """
        #文件路径位于该images目录下
        #该目录下会创建对应类的文件夹，然后按时间进行分类
        #命名规则：调用方法+时间

        # 类名称
        dirname = name

        #确认文件夹是否存在,不存在则创建该文件夹
        if not os.path.exists("../images/"+dirname):
            os.mkdir("../images/"+dirname)

        #有分类文件夹后，再根据时间做出管理
        #例如：确认时间是2025-08-06-17-24-36
        #文件具体命名为：调用方法+20250806172436
        dirname2=datetime.datetime.now().strftime("%Y-%m-%d")

        #确认文件夹是否存在,不存在则创建该文件夹
        if not os.path.exists("../images/"+dirname+"/"+dirname2):
            os.mkdir("../images/"+dirname+"/"+dirname2)

        #创建文件
        filename = sys._getframe().f_back.f_code.co_name+"-"+datetime.datetime.now().strftime("%Y%m%d%H%M%S")+".png"
        self.driver.save_screenshot("../images/"+dirname+"/"+dirname2+"/"+filename)

BlogDriver = Driver()